from PyPDF2 import PdfReader, PdfWriter
import os
import glob

'''
PyPDF2模块：强大的pdf处理模块
PdfFileReader 可以理解为读取器。
PdfFileWriter 可以理解为写入器。

读取器将所有 PDF 读取一遍。
读取结束后，读取器将读取的全部内容交给写入器。
写入器统一输出到一个新 PDF。
'''

goal_pdf_path = 'poetry.pdf'

pdf_writer = PdfWriter() #把写入器实例化

for file in glob.glob('./*.pdf'):
    pdf_reader = PdfReader(file)
    for page in range(pdf_reader.getNumPages()):
        pdf_writer.addPage(pdf_reader.getPage(page))

with open(goal_pdf_path,'wb') as out:
    pdf_writer.write(out)